package com.snopy.leetcode.index1_1000.index101_200;

import com.snopy.leetcode.TreeNode;
import com.snopy.utils.DataFactoryUtil;

/**
 * @author snopy
 * @version 1.0.0
 * @classname Question110
 * @description 平衡二叉树
 *
 * 给定一个二叉树，判断它是否是高度平衡的二叉树。
 *
 * 本题中，一棵高度平衡二叉树定义为：
 *
 * 一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
 * @email 77912204@qq.com
 * @date 2022/04/22 23:06
 */
public class Question110 {

    public static void main(String[] args) {
        Integer[] nums = new Integer[]{1,2,2,3,null,null,3,4,null,null,4};
        TreeNode root = DataFactoryUtil.buildTree(nums);
        int depth = depth(root);
        System.out.println(depth);
        int ldepth = depth(root.left);
        System.out.println(ldepth);
        int rdepth = depth(root.right);
        System.out.println(rdepth);


        boolean balanced = isBalanced(root);
        System.out.println(balanced);
    }
    public static boolean isBalanced(TreeNode root) {
        if (root==null){
            return true;
        }

        return Math.abs(depth(root.left)-depth(root.right))<=1 && isBalanced(root.left) && isBalanced(root.right);
    }
    public static int depth(TreeNode root){
        if (root == null){
            return 0;
        }
        return Math.max(depth(root.left),depth(root.right))+1;
    }
}
